Cleaned up a little thing