Fixed return type which was causing compile error.