Should fix issue with socket not working