Cleaned up something I missed