Attempting a small fix