quick fix, test code