Fixed small problem with types