Fixed bug after stress testing