Fixing bad return type on method