@emir-hasanbegovic added small fix to types