quick fix after review