diff --git a/public/images/icon-exit.svg b/public/images/icon-exit.svg
new file mode 100644
index 0000000..0ffae97
--- /dev/null
+++ b/public/images/icon-exit.svg
@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="white" width="18px" height="18px"><path d="M0 0h24v24H0z" fill="none"/><path d="M19 6.41L17.59 5 12 10.59 6.41 5 5 6.41 10.59 12 5 17.59 6.41 19 12 13.41 17.59 19 19 17.59 13.41 12z"/></svg>
\ No newline at end of file
diff --git a/src/components/cards/RealStockCard.vue b/src/components/cards/RealStockCard.vue
index 3c3fd2f..465cccf 100644
--- a/src/components/cards/RealStockCard.vue
+++ b/src/components/cards/RealStockCard.vue
@@ -10,7 +10,7 @@
             <a href="https://td2.info.pl/profile/?u=17708" target="_blank">Railtrains997</a>
           </h1>
           <button class="btn exit-btn" @click="store.isRealStockListCardOpen = false">
-            &Cross;
+            <img src="/images/icon-exit.svg" alt="" />
           </button>
         </div>
 
@@ -100,13 +100,13 @@ import { useStore } from '../../store';
 import imageMixin from '../../mixins/imageMixin';
 import stockMixin from '../../mixins/stockMixin';
 
-import { IRealComposition } from '../../types';
+import { IRealComposition, VehicleGroupType } from '../../types';
 
-function getVehicleType(stockType: string) {
-  if (/^E/.test(stockType)) return 'loco-e';
-  if (/^S/.test(stockType)) return 'loco-s';
+function getVehicleType(stockType: string): VehicleGroupType {
+  if (/^E/.test(stockType)) return 'loco-electric';
+  if (/^S/.test(stockType)) return 'loco-diesel';
 
-  return 'car-passenger';
+  return 'wagon-passenger';
 }
 
 export default defineComponent({
@@ -223,12 +223,10 @@ export default defineComponent({
 @import '../../styles/global.scss';
 
 .exit-btn {
-  font-size: 1.2em;
-  margin: 0.25em 0;
-}
-
-.btn {
-  background-color: #444;
+  display: flex;
+  background-color: #333;
+  border-radius: 0.25em;
+  padding: 0.5em;
 }
 
 .card_content {
diff --git a/src/types.ts b/src/types.ts
index 848562a..a1f3724 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -3,7 +3,7 @@ export type StockSectionMode = 'STOCK_LIST' | 'STOCK_GENERATOR';
 
 export type LocoGroupType = 'loco-electric' | 'loco-diesel' | 'unit-electric' | 'unit-diesel';
 export type WagonGroupType = 'wagon-passenger' | 'wagon-freight';
-export type VehicleGroupType = LocoGroupType & WagonGroupType;
+export type VehicleGroupType = LocoGroupType | WagonGroupType;
 
 export interface IVehicleProps {
   type: string;