tried a small fix