attempting fix for sklearn