Fixes

Small fix stuff