hopefully fix a few issues
small bug fix