quick fix, may not work :)