cleaning up one more bug