Fix for NoneType object error