Moving towards better gpu memory management
Changed test