Another quick fix for error