added fix for stupid bug