Adding small fix function