Small fix to get rid of null pointer error.