Fixed small test issue with nil