fixed bug where it didnt work for complete mode